package leetcode;

import java.util.HashMap;
import java.util.Map;

/**
 * 138.复制带随机指针的链表
 * 给定一个链表的头节点，该链表除了有next的指向外，还有一个random的随机指向。现在需要你复制一条一模一样的链表，并返回新链表的头节点。
 *
 * 输入：head = [[7,null],[13,0],[11,4],[10,2],[1,0]]
 * 输出：[[7,null],[13,0],[11,4],[10,2],[1,0]]
 */
public class Num_138 {

    private class Node {
        int val;
        Node next;
        Node random;

        public Node(int val) {
            this.val = val;
            this.next = null;
            this.random = null;
        }
    }

    public Node copyRandomList(Node head) {
        //判断为空的情况
        if(head == null){
            return null;
        }
        //建立哈希表，存放旧链表的节点和新链表的节点，一一对应关系
        Map<Node,Node> map = new HashMap<>();
        //开始遍历旧链表，并放map中添加节点元素
        for(Node x = head; x != null; x = x.next){
            Node node = new Node(x.val);
            map.put(x,node);
        }
        //再次遍历旧的链表，根据旧链表的指针关系，设置新链表的指针关系
        for(Node x = head; x != null; x = x.next){
            map.get(x).next = map.get(x.next);
            map.get(x).random = map.get(x.random);
        }
        //返回新链表的头节点
        return map.get(head);
    }
}
